Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

remove check_storage_aliases #1451

Merged
merged 7 commits into from
Dec 5, 2024
Merged

remove check_storage_aliases #1451

merged 7 commits into from
Dec 5, 2024

Conversation

jjsjann123
Copy link
Collaborator

@jjsjann123 jjsjann123 commented Nov 18, 2024

The motivation is to remove the use of numel in code logic, since number of elements doesn't play well with dynamic shape yet.

I don't know what check_storage_aliases is protecting us against. So I removed the whole thing and looks like we don't have any test validating the logic neither.
This is a follow up on conversation here: https://github.com/Lightning-AI/lightning-thunder/pull/877/files#r1844732984
cc'ing @crcrpar for visibility.

thunder/__init__.py Outdated Show resolved Hide resolved
@jjsjann123 jjsjann123 changed the title refactor to remove use of TensorProxy.numel [DO NOT REVIEW] refactor to remove use of TensorProxy.numel Nov 19, 2024
@jjsjann123
Copy link
Collaborator Author

failing tests seems to be coming from nvfuser.

thunder/__init__.py Show resolved Hide resolved
thunder/__init__.py Show resolved Hide resolved
@jjsjann123 jjsjann123 changed the title [DO NOT REVIEW] refactor to remove use of TensorProxy.numel remove check_storage_aliases Dec 1, 2024
@jjsjann123 jjsjann123 marked this pull request as ready for review December 1, 2024 21:28
@jjsjann123 jjsjann123 requested a review from mruberry December 4, 2024 22:10
@mruberry
Copy link
Collaborator

mruberry commented Dec 4, 2024

@jjsjann123 Is this ready for a review now?

@jjsjann123
Copy link
Collaborator Author

@jjsjann123 Is this ready for a review now?

Yes. Since this is only removing code +0 -40. I didn't ended up putting any comment in.

Copy link
Collaborator

@mruberry mruberry left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@mruberry mruberry enabled auto-merge (squash) December 5, 2024 19:31
@mruberry mruberry merged commit 02085e1 into main Dec 5, 2024
42 checks passed
@mruberry mruberry deleted the removing_use_of_numel branch December 5, 2024 19:34
@crcrpar crcrpar mentioned this pull request Dec 19, 2024
riccardofelluga pushed a commit that referenced this pull request Jan 27, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants